# model settings
# Only for evaluation
model = dict(
    type='Sep_ImageClassifier',
    backbone=dict(
        type='SwinTransformer',
        arch='base',
        img_size=384,
        stage_cfgs=dict(block_cfgs=dict(window_size=12))),
    neck=dict(type='GlobalAveragePooling'),
    head=dict(
        type='MultiLabelLinearClsHead', num_classes=1000, in_channels=1024))
# loss=dict(type='CrossEntropyLoss', loss_weight=1.0),
# topk=(1, 5)))
# custom_imports = dict(
#     imports=[
#         'medfmc.datasets.medical_datasets',
#     ], allow_failed_imports=False)
